tensor([[[[[0.8300, 0.7400],
           [0.0000, 0.8300]],

          [[0.6300, 0.0000],
           [0.5500, 0.0000]]],


         [[[0.0600, 0.0000],
           [0.5500, 0.0000]],

          [[0.2500, 0.3000],
           [0.0000, 0.2900]]]],



        [[[[0.0000, 0.0000],
           [1.0000, 0.0000]],

          [[0.0000, 1.0000],
           [0.0000, 1.0000]]],


         [[[0.9400, 0.9200],
           [0.0000, 1.0000]],

          [[0.6300, 0.5800],
           [0.6100, 0.6700]]]],



        [[[[0.1700, 0.2600],
           [0.0000, 0.1700]],

          [[0.3400, 0.0000],
           [0.3400, 0.0000]]],


         [[[0.0000, 0.0800],
           [0.3400, 0.0000]],

          [[0.1300, 0.0000],
           [0.2400, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0300, 0.0000],
           [0.1100, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.1100, 0.0000]],

          [[0.0000, 0.1200],
           [0.1500, 0.0400]]]]], grad_fn=<RoundBackward1>)